package com.xbh.management.utils;

import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

/**
 * @program: management-center
 * @description:
 * @author: 许宝华
 * @create: 2021-09-22 16:45
 */

public class FileUploadUtils {

    /**
     * 随机生成图片名
     * @param file
     * @param dirPath
     * @return
     */
    public static String uploadImage(MultipartFile file, String dirPath){
        //判断该目录是否存在，如果不存在则创建目录
        File folder = new File(dirPath);
        if (!folder.exists()) {
            folder.mkdirs();
        }
        // 获取原文件名称
        String oldName = file.getOriginalFilename();
        // 获取原文件的后缀，并生成新文件名
        String newName = UUID.randomUUID().toString() + oldName.substring(oldName.lastIndexOf("."));
        try {
            //将文件上传到服务器
            file.transferTo(new File(folder, newName));
        } catch (IOException e) {
            e.printStackTrace();
        }
        return newName;
    }

    /**
     * 删除文件
     * @param file
     */
    public static void deleteFiles(File file){
        if(file.exists()){
            file.delete();
        }
    }

}
